import { Injectable } from '@nestjs/common';
import { TEMPLATE_CATEGORIES, BASE_TEMPLATES } from './base-templates.config';
import { TemplateCategory, BaseTemplate } from '../types/template.types';

@Injectable()
export class TemplateService {
  // 获取所有模板类型
  getCategories(): TemplateCategory[] {
    return TEMPLATE_CATEGORIES;
  }

  // 获取基础模板
  getBaseTemplate(category: string): BaseTemplate {
    return BASE_TEMPLATES[category];
  }

  // 获取所有基础模板
  getAllBaseTemplates(): Record<string, BaseTemplate> {
    return BASE_TEMPLATES;
  }

  // 根据类型获取参数schema
  getParamSchema(category: string) {
    const template = this.getBaseTemplate(category);
    return template?.param_schema;
  }

  // 获取工作流JSON
  getWorkflowJson(category: string) {
    const template = this.getBaseTemplate(category);
    return template?.workflow_json;
  }
} 